/**
 * 版权所有 2009-2012山东新北洋信息技术股份有限公司
 * 保留所有权利。
 */
package com.linyaonan.leetcode.easy._501;

import lombok.val;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

/**
 *
 * 给定一个有相同值的二叉搜索树（BST），找出 BST 中的所有众数（出现频率最高的元素）。
 *
 * 假定 BST 有如下定义：
 *
 * 结点左子树中所含结点的值小于等于当前结点的值
 * 结点右子树中所含结点的值大于等于当前结点的值
 * 左子树和右子树都是二叉搜索树
 * 例如：
 * 给定 BST [1,null,2,2],
 *
 *    1
 *     \
 *      2
 *     /
 *    2
 * 返回[2].
 *
 * 提示：如果众数超过1个，不需考虑输出顺序
 *
 * 进阶：你可以不使用额外的空间吗？（假设由递归产生的隐式调用栈的开销不被计算在内）
 *
 *
 * @ProjectName: leetcode
 * @Package: com.linyaonan.leetcode.easy._501
 * @ClassName: FindMode
 * @Author: linyaonan
 * @Date: 2019/12/31 16:25
 */
public class FindMode {

    public int[] findMode(TreeNode root) {

        if (root == null) {
            return new int[0];
        }
        // 用于存放
        Map<Integer, Integer> map = new HashMap<>();
        getTreeNode(root, map);
        LinkedList<Integer> saveList = new LinkedList<>();
        int max = 1;
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            Integer value = entry.getValue();
            if (value == max) {
                saveList.add(entry.getKey());
            } else if (value > max) {
                saveList.clear();
                saveList.add(entry.getKey());
                max = value;
            }
        }
        int[] r = new int[saveList.size()];
        for (int i = 0; i < saveList.size(); i++) {
            r[i] = saveList.get(i);
        }
        return r;
    }

    private void getTreeNode(TreeNode node, Map<Integer, Integer> map) {
        if (node == null) {
            return;
        }
        int val = node.val;
        saveNodeOne(val, map);
        getTreeNode(node.left, map);
        getTreeNode(node.right, map);
    }

    private void saveNodeOne(int val, Map<Integer, Integer> map) {
        if (map.containsKey(val)) {
            map.put(val, map.get(val) + 1);
        } else {
            map.put(val, 1);
        }
    }
}
